package top.jacktgq.imooc.candychat;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

public class WebSocketServer {
    public static void main(String[] args) {
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        
        ServerBootstrap bootstrap = new ServerBootstrap();
        
        bootstrap.group(bossGroup, workerGroup)
                .channel(NioServerSocketChannel.class)
                .childHandler(new ChannelInitializer<Channel>() {
                    @Override
                    protected void initChannel(Channel ch) throws Exception {
                        // WebSocket基于Http协议，所以要有Http编解码器
                        ChannelPipeline pipeline = ch.pipeline();
                        pipeline.addLast(new HttpServerCodec());
                        // 对写大数据流的支持
                        pipeline.addLast(new ChunkedWriteHandler());
                        // 对HttpMessage进行聚合，聚合成FullHttpRequest或FullHTTPResponse
                        // 几乎在基于Netty的WebSocket的编程中都会用到此Handler
                        pipeline.addLast(new HttpObjectAggregator(8192));
                        
                        //================以上是用于支持Http协议的Handler=================
                        
                        /**
                         * 1、对应WebSocket，它的数据是以帧（frame）形式进行传递的
                           2、可以看到WebSocketFrame下面有6个子类
                           3、浏览器发送请求时，ws://localhost:7000/xxx 表示请求的uri
                           4、WebSocketServerProtocolHandler 核心功能是将http协议升级为ws协议，保持长连接
                           5、是通过一个状态码 101将协议进行升级
                              WebSocket服务器处理的协议，用于指定给客户端连接访问的路由：/ws
                                                                                         本Handler会帮你处理一些繁重复杂的事：
                                                                                             （1）会帮你处理握手动作：handshaking(close, ping, pong) ping + pong = 心跳
                                                                                             （2）对应WebSocket来讲，都是以frame进行传输的，不同的数据类型对应的frame也不同。
                         */
                        pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
                        
                        // 自定义Handler
                        pipeline.addLast(new ChatHandler());
                    }
                });
        try {
            ChannelFuture future = bootstrap.bind(9999).sync();
            future.channel().closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }
}
